Add more comments in the JS