Update experiment.py

Changed to a get method