small changes on the JS file