Add argument parser to the script