Minor improvements to visualization, bug fixes.