add code for homeworkDuan