Update README.md

Update setup example and TODO