Cleaning up coding style in functions.php