clean up
added documentation