xuefei wang's python hw1 source code