Update remove_zeros.py