Update README.md

Add future work to be done for this project