added missing args to main func