Thanks Pete.
I looked at the Lionel parts from another post. I thought some would work. I have no experience with the steam engines and didn't know what was available.
In the past when I've ordered parts from Lionel, they arrived very fast. I've even got a few extras like a new auto carrier from their parts page. So if I knew what to get, I would order again.
Right now, I'm thinking a motor mount and the flywheel would do. My homemade dog bone doesn't look great. It does work well and nobody will see it.