Add command.py with argparse