Recreated the treaty feature