Learning to reason for long-form story generation