Update diag.rb