work on story mode