Updated the program to produce prettier output