TASK DEFINITION: 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.
PROBLEM: Premise: Mahlon has ADHD and often fidgets and goofs off at school. 
 Initial Context: Today Mahlon was sucking on the conical metal tip of a pen. 
 Original Ending: Mahlon had the metal cone in his teeth and was blowing in and out. Mahlon was making an annoying whistling sound with the pen tip. Mahlon didn't stop until he inhaled too hard and almost swallowed it. 
 Counterfactual Context: Mahlon was suspended from school for disrupting class.

SOLUTION: Mahlon had a metal cone in his teeth and was blowing in and out. Mahlon was making an annoying whistling sound with the pen tip. Mahlon didn't stop until he inhaled too hard and almost swallowed it. 

PROBLEM: Premise: Miley was trying to charge her cell phone. 
 Initial Context: The connector was getting sort of loose and it didn't charge well. 
 Original Ending: Finally she decided to order a new charger online. It arrived within a couple of days. She loved it, it charged her phone much better. 
 Counterfactual Context: The connector allowed the fastest charging she ever had.

SOLUTION: She decided to order online the same charger for her mom. It arrived within a couple of days. Her mom loved it, it charged her phone much better. 

PROBLEM: Premise: David went to the mall. 
 Initial Context: When he got home he realized his wallet was missing. 
 Original Ending: David called the mall and ask if it had been found. They said it was turned in so he went to pick it up. All the money was gone, they had stolen it prior to turning it in. 
 Counterfactual Context: Suddenly, his mom called him to go back home.

SOLUTION:
David returned home to call the mall and ask if it had been found. They said it was turned in so he went to pick it up. All the money was gone, they had stolen it prior to turning it in. 

