Updated experimentation.py