Merge pull request #10 from mattybeam/master

First two steps of EM algorithm