Cleanup
Add docs