Cleaned up
Updted docs