add function for check the screen size