updated python notebook for linear regression