Atlas 0-8-0 n scale steam with tender

This unit has the electrical pick up from the tender wheels, and conducts to a pin on the back of the engine.
This is ok until you use this on a n trak modular layout with blocks, at the end of one block , I lose contact.
I thought of manufacturing a copper brush pick up on the front, any suggestions?