new file:   problem6.py