Changes to use IMU data