I have a Hershey engine from a set and it puffs like crazy. I just bought its smoke parts from Lionel and going to see if I can tinker and get it in the blue comet or southern and see how it does. The only thing the Hershey smoke unit doesn't have is a spot for the front light.
worse comes to worse I simply put the original smoke unit back in and leave it alone.