- additional documentation
- clean up code!
- TESTS