ADD: More documentation

Cleanup code
Merge remote-tracking branch 'origin/master'