Add this code to functions.php: