Bug fixes and code improvement

Changed white key artifact to restart the game.