adding arg parsing