Added TODO and changes to README file