Units now move after battle