Add pyyaml dependency to setup.py
Change argparse actions to store instead of nargs