Update collaborative_filtering.py