Fix unit moves highlighting

If a unit does not use up all its moves in one movement, the next time
it is picked up, the squares highlighted will reflect that.