'Small changes to the display code (still learning)