Added some more test code to test the display