Fixed import error in similarity.py