Add configuration for state size