First working version of the classifier