Update README

Added TODO part