Add some comments to visualizer js