Fixing code and adding more info to README.md