Simplify last edit to functions.php