update diag class