continue UI improvements