add more features for classification