Remove "other" column for simplicity during training