added ability to restart level