Cleaned up code and added comments
Add notes on original classes for the assignment