some ui polishing.