Initial commit of Problem1.py, working