linear regression from scratch python code