Update testCode.py