new look

new look to the game with shorter code.