Fixed #10 and made the screen size bigger!