Correlation in Regression — Interview Query machine learning problem