Adding the movement code