HW1 - KNN model on iris data