added token button behaviour