Changing print statement to logging