Adding model classes to the project