Add primary key for migrations