fixed UI bugs