Task: Story Composition
Input (English): Premise: Tim was terrified of swimming. 
 Initial Context: His parents desperately wanted him to learn to swim. 
 Original Ending: Tim's parents signed him up for swimming lessons at the local pool. At first, Tim was too scared to get into the pool. After a month, Tim was swimming on his own and started to enjoy it! 
 New ending: Tim's parents signed him up for swimming lessons at the local pool. At first, Tim was too scared to dive into the pool. After a month, Tim was diving on his own and started to enjoy it! 
Output (English): His parents wanted him to learn how to dive.