adding argparse

for better parsing of args