Added a TODO
Slight style improvements