General bug fixes and overall UI improvements.