Add better comments to the JavaScript