nearly working number of tokens not quite right yet