Update main.yml

Changed from model "yum" to module "apt" for ubuntu