Update comments in solver.py