Modify argument

if user don't input argument, enter the value from console