Added the Dirichlet Process to estimate K