My homework in studying python