Update movement code