Cleaned up functions.php