Create functions.php

Add this code to the theme's functions.php