Added UI improvements.