UI updates and misc functionality