Add additional movement code