Added more feedback and error checking. Able to classify, although not correctly.