Updates to visualization