Add partitioned by annotation