feat(stage2.py): Starting to automate the tests to run