Updated class diagram for current software version