Created some sample data and made test for it