Clean up experiment.py
Move script sections to imitate references to reference.py