step 4 - model diag