added trivial print-diagonal code