Improved search fonction