Initial working commit of problem6.py