updates to heat example