Updated main.yml