added tools to train and test classifier