Added code to the JS file