improvements to visualization