Add print of the diameter at the end of main