improved UI

Signed-off-by: Anton Chan <antoniuschan99@gmail.com>