Updated README example and TODO