added correlation plot