Add rate limit check