Some changes in IMU code