Started working on game states