new features: selecting specific columns for training