Works for the 4 assemblage test case!