UI improvements.