Update compute_Nz.py