Added CLI arguments for run