updating task 2, fixing indexes for features