started adding feature extraction code