fix bugs in input folder name, add input argument to the code