Updating readme

Adding TODO