Update baselines.py