Updated README and removed TODO