Added test data class