Added dataset X loading for testing