Added the code to the theme