push the function into functions.php