Add check for screen size