minor updates to task2.py  - printing results