Some minor changes to the JS file