Code is now working
Updated README.md file