add menu screen and add restart button