Add classes for Kd-Trees assignment