Add diagram class for header diagram