update the figures and kNN code
update the graphs