refactor UI class