Add missing funtion that sets the rotation