Added a new Train class and did some extra work on sample_vehicle