Add TokenBucket code.
Add test cases.