modified code to run heat stake test