hw2 part 1 find_track modification