regression.py updated