Put query functions back in functions.php