Using chi-squared for important features