changed bug in setup.py, added tree-option for plot_heatmap