Add ability to register a primary key. Pair with @LiGang9090.