adding class diagram and class definition