more changes to classification script to determine optimal number of features