expanding classification code to try out different classifiers