Fixing small ui issues.