Added ability to get screen size of display