Updated the README to reflect changes in code