improving layers that go into heatmaps