More updates to visualization