Add UI fixes.