
- tests
- more documentation
- more testing
- look into results/class mapping

