Add class diagram to represent the entities of program