Add block to the formals of the function in with_block