Improve short story section