Adding small UI update