Improved rate limit checking