Add TODO's
Remove unused function