How is Na concentration re-established after action potential
Yes, there are Na-K pumps all over.

However, it is important to recognize that ion concentrations change very little during action potentials. The change in membrane potential is due to changes in ion conductances, very few ions actually flow. It is still necessary to maintain concentration gradients over many many action potentials, but the change in concentrations during the course of a single action potential is negligible.