Move code from after a return statement to before it