HW3. Part 3 setup