cleanup
Adds small javadocs in code