change feature selection logic