Use argument as input if provided