Move some images from test to train
Add KNN classifier