added the nice correlation plot