improving cosine sim