Cleaning up the JS file