Almost finished with feature selection