Began on Class Diagram