Add diagnostics output