Add audit logging functionality