UI - updates

Signed-off-by: Thomas Yuill <thomas.yuill@gmail.com>