Added a basic drive train
Merge pull request #2 from itsZN/master

Updated Drive train code