Updated regression.py