couple of small UI fixes