Updating to add CLI arguments