added part2.py