Given a premise, an initial context, an original ending, and a counterfactual context, the task is to generate a new story ending aligned with the counterfactual context and as close to the original ending as possible. Each instance consists of a five-sentence story. The premise is the first sentence of a story, and the second sentence, which is the initial context, provides more information about the story's context and the story's general plot. The original ending is the last three sentences of the story. Also, a counterfactual context is a slight modification to the initial context. You should write a new story ending that edits the original story ending as little as possible to regain coherence with the counterfactual context. To sum up, you should write the last three sentences of a story based on the premise(first sentence) and the counterfactual context(second sentence) of the story.

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.
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. 

Premise: I said a prayer for Michael earlier today. 
 Initial Context: He was sick in the hospital. 
 Original Ending: A lady gave Michael AIDS. I prayed hard for Michael today. Michael just died. 
 Counterfactual Context: He was my pet turtle.
A lady gave Michael poison. I prayed hard for Michael today. Michael just died. 

Premise: Tim was trying to draw the perfect circle. 
 Initial Context: He could not do it. 
 Original Ending: He tried twelve times. He had an idea. He traced around a CD to get a perfect circle. 
 Counterfactual Context: He could do it repeatedly with ease.

He did it twelve times. He traced around a CD to get a perfect circle. He compared that circle to his circles. 

