Create HW4_amitkal_p234.py