- cleaned up code
- fixed failing tests
- added basic class diagrams