Disables keys after game-end and adds restart functionality