UI update - remove duplicate functionality