Add diagram class