Create yat_hw1.py

This is my HW1 in python