Fix for hw3-part2