Final UI fixes