Added Correlation plot and Train/Test with Chi2 and KS test