Update read_sol.py