Refactor functions.php to use theme setup function