Add logic to check for token.