Updated dataset and notebook