Update path to dataset
Add documentation