add the homework4 implementation still need test….