Update logging in ImportanceDiffusionAgent