Getting JS to work with the theme