update docstring and add methods to return tokens.