Fixed few UI bugs.