UI updates for new functionality