update, minor UI improvement