Updated README to match updated code