adding class diagram and data structure