Updated audit code