Fixed functions.php for prod