Update DriveTrain.java

Fixed comments, made my stuff a lot better.