Update README

- add a few TODOs in the code