Cleaned up and debugged new experiment.py