added graphs, made some changes that should probably be undone to kmeans_warm_up.py