Update index w HW4 code