update README to match notebook