Add documentation
A little clean up