Removed old code
Removed old code.
Updated class diagram
Comments added
Added comments