Remove files
Cleanup tree + add tests
Add tests and docs