Update README.md

Updated the README to provide more information about the code.