UI changes, bug fixes.