Updating README to reflect the latest code