miss some modfication; add doctest in a2.py for futher test