K-nearest neighbours for the AI course