add stuff for heat maps - not ready