Add audit logging