UI design update