Updating TODO and docs