Final changes to visualization