Add print statements to the code to see the output

Add print statements to the code to see the output #14