Added code to TrainWagon/Truck