First linear regression in Python