Working on game logic