UI improvement
2 files changed