Modify to need config.yaml

Modify to need config.yaml