updated to use functions.php