Add done check and UI for level completion