added two print statements to the end