Small changes to the JS