added diag classes