Fixing the finally block
Adding in the finally blocks