add ui improvements