fixing up storyMode code