small UI fixes
small UI fixes