UI fixes and documentation