Adding class diagram