Minor code reformatting in functions.php