* First commit * isort * black * tweak prompt * fix for argparse * fix typo * add e2e * Add test files, clean * fix black settings * revert * fix trailing * remove extra * comment * small fix, updated prompt * Fix argparse * small improvements * fix for merge * fix for merge