Python Expectation Maximization