Added TODO and updated README