fixing caption of heat maps