Create HW4_amitkal_p1.py