- Refactored code
- starting to add logistic regression (stub)