Tests pass. Minor update to API as we need to specify how many samples we want in the case of an erasure.