Addded more marginals tests and fixed assignments test.