Add check for screen width