Redid the Class Diagram