Combined story functionality into one class