checkpoint
add cosine similarity plotting script