Added Schematic for train, cleaned up some code.