Need to set primary key as well