ui updates, bug fixes