Add command line input and output