more updates for ui