Cleaned up functions.php file