UI improvements with test files