Added in flatten tests