Split test data into own class