add unit swapping in battle prep