Implement linear and logistic regression | Uber Interview Question