functions.php fixes