add diag plot