Update main.yml

Adding TODOs