Adds dataset functionality
Continues adding dataset functionality