Add logging modifications