Added running on test data