Added n_neighbors for plotting