Add try/finally block