Update function.php

Add loading of the custom.js file