Added logic to stage the files after creation