Add class diagram to design