Add rate limiting