Update functions.php to comply with Theme Check