Yes, add the code in functions.php file of the activated theme.