a few changes to heatmap.py, some testing for various heatmap functions