Update experiment.py

Talking to Laurence, decided to move this to be a more generic resource