[IMP] write in experiment.py