improve UI of TOS

Signed-off-by: Thad Allender <thadallender@gmail.com>