Shrink size of display
Add main method to test