Updates to visualization scripts.