Cs461 HW; Going back to complete #5