TODO updated to readme