Update regression.py