Add two input arguments for better automation.