Fixed script in main in ThermalProblem