Updated the README to reflect the new code