Cleaning up documentation
Refactored and updated docs