Remove unnecessary code in functions.php