Create thermal_model.py