Update experiment.py

This is a more clear way of doing this