Adding in EM Algorithm