Add k-NN and output analysis to experiment