modified python a.py