small improvements to visualization