Update print statements in main method