UI updates bug fixes