finished bit about feature extraction