Add security checks