Update BHZ and wave functions