minor ui polish