wrote some more and generated proper heat maps