functions.php code improvements