Adding a function to create test data