Fixes #9 - removes TODO.