better UI

Signed-off-by: AndrewPalmer <a.d.palmer@gmail.com>