Minor changes in functions.php