Add python script to plot correlation matrix