Update ps6_test.py