I am actually the creator of the original notebook. Happy to do more, do you know of any good datasets for these that would be a good starting points?