Move code reloading around