Removed unnecessary UI code and changed design. Added comments on each class.