HW1 Python script