just tweaking the arg parsing