Update task10.py

Need this too