Add menu and improved the reset game method