Update UI

Fix script