Fixed the solver diagnostics example to work with the new
improve_candidates argument