Update testcode.py