Update ZSG_solver.py