A simple recommendation system in python