added model class