Modified the way we update the cnxnPartition map