WIP: Segfault is fixed, KNN is run after EM loop