Add IMU init and reset in Python side