Updated README to reflect changes in the code