Added automated feature generation, removing highly correlated features, and plotting ROC curve