Updated README to be up to date with code