How Willy Wonka and the Chocolate Factory should have ended