Make primary_key configurable