added code to plot the electrostatic potential vs Z axis