added figures, class diagrams