Heatgen script is now working