Implemented changes for shift to cosine similarity graph