Add feature_count routine