Fixes #56
Updated TODO