Updated code for audit logs