UPDATED: Enable to input a file as an argument.