Update chomp_solver.py