A couple UI fixes.