updated visualization scripts