Updated unit_test2.py