UI improvements

Signed-off-by: Clienthax <clienthax@gmail.com>