Updated UI bugs fixes