Added test for Data class