Now makes an image like in HW4 using the k-d tree