Tidy up story buttons