Documentation updates

Working on documentation and general clean up