put this in functions.php