Add class diagram