Add feature to specify config keys