added tests replicating our paper