Final commit problem6.py