Fix Up functions.php