Create ps7.py

finish first 1-4 problems