The homework code for DataStructure