The lithium beam is an effective diagnostic tool for investigation of stability and particle transport in the pedestal. It was used successfully to measure edge current density on DIII-D, achieving qualitative agreement with neoclassical models. Electron density profiles were also measured. Proposed upgrades will continue these measurements with higher reliability as well as explore edge current measurements using spectroscopy. The optics will be redesigned to optimize throughput and aperture broadening and to replace the photomultiplier tubes with avalanche photodiodes. The new system will yield detailed measurements of the pedestal, complementing existing diagnostics for investigating pedestal stability, edge localized mode cycle, and particle transport through the pedestal.