Add try/catch block