Added TODO and better readme