Update network_annealing.py