Adding function to test data