Basic template for HW3 with kmeans somewhere there , and EM