Wrap all code in the same closure