Added code.wrap