Updated Readme

- Added TODO
- Slight improvements to example