simple linear regression in machine learning python code