Adding print statements to main method