work on story mode and level select