Task: Story Composition
Input (English): Premise: Kelly was bathing her baby. 
 Initial Context: The baby did not like the water. 
 Original Ending: He cried and screamed the entire time. Kelly cleaned him as quickly as possible. He finally calmed down once she dried him off. 
 Counterfactual Context: The baby loved the water.
Output (English): He laughed and smiled the entire time. Kelly cleaned him as slowly as possible. He finally calmed down once she dried him off.