Bug fixes and improvements of translation UI