updated UI bugs