Fix TODOS

Fixes #4