Update network_solver.py