Add movement and rotate function