Updated code for IDEAS exercise

Check heating system control for faster simulation