Working version of EM, should be able to run it on test dataset. Might need to change the way we pass input data to class