Update compression_solver.py