Improve item assist UI