improve ui (a bit)