add diag code