Fixed some ui issues.