First attempts at heat map, but stuck at the moment