Added framework for M-Step. More work coming