Adds 2 more model classes