Add movement code