Modification to how heat maps are calculated