[mod] refactor the argument input