An implementation of K-Nearest Neighbour written in Swift