test suite and UI improvements