split up code into functions, user argparse
prototype setup.py