Model classes added

Problem
Problem solution