little UI fixes