update to wavefunction plots (HO and primarily Mom