updated test class for new data