Adding model class