Add missed print function to problem1