Swap code to be in a code block