adding to function.php