Updated moving logic for offensive units in all stages