Fixing bug with printing columns and rows of tokens