Add identity key type parameter to Store