small UI fixes