Inertial Sensor and Premade Drivetrain

Yes, the concept is correct in general.

However you cannot mix and match controlling motors individually and with drivetrain class.

See this topic for explanation of the differences between drivetrain and smartdrive classes:

Also, here is the API fir the smartdrive class:

Finally, if you choose not to use smartdrive with its built-in PID, here is a PID tutorial: