HW4 index fixes