added argparse

annoyingly it messes up unit test