Added code for sample data generation.
Also made the draft story logic work. Now it does find draft story.