Working on adding command line arguments