updated IMU code