Updating functionality in the README.md