Started to add some logic to the game