fixed ui bugs