Code for the Yelp dataset challenge