extended this file
HW3, part 3