Added argparse and now works