Fixed UI bugs