Updating Model class. Adding polymorphic associations