Small update for the UI

Signed-off-by: Jozsef Kiraly <fonix232@gmail.com>