Update main.yml

Some minor changes