Inital working commit of Problem3.py