simple linear regression implementation in python