Fix comment that reference heat map example