Replace dataset with distribution

I'm not entirely sure but this looked like a copy-paste gone wrong. Side note, it would be nice to also have a diagram with all the classes and their relationship in this document.