updates in regresion.py