I have a very old 0-6-0 switcher in great shape with the upright flat motor, but the drive wheels in the center have come loose, and need replacement. Will the older first design listed in the parts section work as a replacement for the drive wheel design of the older model. It ran fine, but now the wheels need to go. Any suggestions?