add partitionId to few models