Add main with arguments