Update src/drive_train/DriveTrain.h

Good job!  See the TODO markings; I understand it's work in progress, but keep plugging away.