Assignment 5 - Kd trees