Added KNN classifier categories test
Updated license and remove unused imports
Switched to unittest format
Switched from print to logger