some homework about Python