made it easier to run test case.  cleaned up imports.  fixed get_display...

..._size.  fixed test example so it runs