Update unit-tests
Cleanup and documentation fixes
Remove JetBrains annotations dependency, fixes #3