Added basic dataset class