Add storiesModel class