feat(stage2.py): Added I2C back in