wap to implement linear regression using python