UI touchups, bug fixes