add heatmap.py, which draw heatmap for a correlation matrix