updated unit movement code