Change Yelp class to a module