Update todo.py

Updating import statement