add diagram class