Added script to attempt to calculate Azz from wavefunctions