- Improve parameter types
- Include logistic estimator
