Trying to fix using 'contains' method