Add tokens field to the model