fixed overfitting and correlations, DR needs to be fixed