Fixed pz

Wasn't actually plotting pz before.
Sum wavefunctions for each atom, not finished

Doesn't take into account angle or coefficient yet.