Add command line argument handling for input / output