Single Linear Regression Python Code With Multiple Functions