Fixed functions.php