Quick bug fixes

Hearts now are re-displayed when the user re-starts the game.