Clean up datasets and add some docs