Improve 10.py