Updated class diagram for figures