Added test for data method