align_plots function

Align decision tree and heatmap: