Modify main.yml