Added the last part 7 to hw2b.py