Initial working commit of Problem4.py