added few things on game logic