Tiny refactor
Added TODO file