Setup and dataset loading