Small change to visualization