Working and tested, need to modify KNN to produce images