edit & add

minor UI fix