Refactor + clean up + document JS visualization code

This commit should not introduce any substantial functionality
differences. It just cleans up the JavaScript side of this patch
such that it is easier to follow.

Author: Andrew Or