Did a little with correlation_plot.py