Update README

Add TODO section