Week9 python code set