Changed input method to CLI argument