Clean up functions.php