initial create

This is my HW1 python code.