Wavefunction plot
Improve and remove legacy code
More robustifications
Fix issues with potential plot scripts