Add correlation plot