update the notebook for the movielens dataset