Added a sample dataset to test metrics and updated testloader class