Add rate limit support