Update UI on stop