Added second function to search functions.php