Fix output to only display tokens