Updated and expanded E-step tests. All should pass now