Add optional per-token caps