Fixed bugs with restart level button.