Implement Token Bucket client
Add initial tests