Task: Story Composition
Input (English): Premise: Sophia wasn't a good swimmer. 
 Initial Context: Sophia was relaxing by the pool with some friends. 
 Original Ending: Her friends played a joke on her and pushed her into the pool. Sophia panicked and was drowning. Her friends had to jump in and save her life. 
 Counterfactual Context: So she decided to ask her friend Jax for swimming lessons.
Output (English): Her friend played a joke on her and pushed her into the pool. Sophia panicked and was drowning. Her friend had to jump in and save her life.