Fixed submitted homework code.