Add Model.B and unit test