fixed ec2 -> E2 in viz.py