Increase experiment.py coverage

Refactor