Add simple heat_test script