Update UnittestRoughPartA.py