fix up the last homework