Python hw1