Fixed the pydy viz stuff to work correctly.