Added a function to functions.php