added type for features in training datasets