Cleaning up functions.php