changes to the visualization portion of the experiment, seem to work now