Update TestModule.py