refactor display code
2 files changed