Add try/except to handle non-convergence