Add function to compute heat color gradation