Minor pep8
Add todo