Added documentation to classes and refactored code