Allows specification of knn gallery name as a parameter 

Also adds a visualization thing