Modified heatmap to use jet colormap

This fixes #1 and close #4.